/*
给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。

请你找到这个数组里第 k 个缺失的正整数。

 

示例 1：

输入：arr = [2,3,4,7,11], k = 5
输出：9
解释：缺失的正整数包括 [1,5,6,8,9,10,12,13,...] 。第 5 个缺失的正整数为 9 。
示例 2：

输入：arr = [1,2,3,4], k = 2
输出：6
解释：缺失的正整数包括 [5,6,7,...] 。第 2 个缺失的正整数为 6 。
 

提示：

1 <= arr.length <= 1000
1 <= arr[i] <= 1000
1 <= k <= 1000
对于所有 1 <= i < j <= arr.length 的 i 和 j 满足 arr[i] < arr[j] 

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-missing-positive-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

#include "../stdc++.h"

class Solution {
public:
    int findKthPositive(vector<int>& arr, int k) {
        int lastMiss{-1};
        int current{1};
        int i{0};
        int n = arr.size();
        for (int missCount{0}; missCount < k; ++current) {
            if (current == arr[i]) {
                i = (i + 1 < n) ? i + 1 : i;
            } else {
                ++missCount;
                lastMiss = current;
            }
        }
        return lastMiss;
    }
};

// 二分
class Solution {
public:
    int findKthPositive(vector<int>& arr, int k) {
        if (arr[0] > k) {
            return k;
        }
        int n = arr.size();
        int left{0};
        int right{n};
        while (left < right) {
            int mid = left + (right - left) / 2;
            int x = mid < n ? arr[mid] : INT_MAX;
            if (x - mid - 1 >= k) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }
        return k - (arr[left - 1] - (left - 1) - 1) + arr[left - 1];
    }
};